# +----------------------------------------------------------------------
# | ChatWork智能聊天办公系统
# +----------------------------------------------------------------------
# | 软件声明: 本系统并非自由软件,未经授权任何形式的商业使用均属非法。
# | 版权保护: 任何企业和个人不允许对程序代码以任何形式任何目的复制/分发。
# | 授权要求: 如有商业使用需求,请务必先与版权所有者取得联系并获得正式授权。
# +----------------------------------------------------------------------
# | Author: ChatWork Team <2474369941@qq.com>
# +----------------------------------------------------------------------
from fastapi import APIRouter, Depends
from hypertext import R, response_json, PagingResult
from apps.admin.schemas.agent.know import dataset_schema as schema
from apps.admin.service.agent.know.dataset_service import KnowDatasetService

router = APIRouter(prefix="/dataset", tags=["知识库数据"])


@router.get("/lists", summary="数据列表", response_model=R[PagingResult[schema.KnowDatasetListVo]])
@response_json
async def lists(params: schema.KnowDatasetSearchIn = Depends()):
    return await KnowDatasetService.lists(params)


@router.get("/detail", summary="数据详情", response_model=R[schema.KnowDatasetDetailVo])
@response_json
async def detail(params: schema.KnowDatasetDetailIn = Depends()):
    return await KnowDatasetService.detail(params.uuids)


@router.post("/delete", summary="数据删除")
@response_json
async def delete(params: schema.KnowDatasetUuidsIn):
    return await KnowDatasetService.delete(params.uuids)


@router.post("/retry", summary="数据删除")
@response_json
async def retry(params: schema.KnowDatasetUuidsIn):
    return await KnowDatasetService.retry(params.uuids)
